<?php

/**
 * @file
 * Filter on country.
 */

class location_handler_filter_location_country extends views_handler_filter_in_operator {

  public function option_definition() {
    $options = parent::option_definition();
    $options['operator'] = array('default' => 'in');

    return $options;
  }

  public function admin_summary() {
    return '';
  }

  public function get_value_options() {
    $this->value_options = location_get_iso3166_list();
  }

  /**
   * Provide widgets for filtering by country.
   */
  public function value_form(&$form, &$form_state) {
    $this->get_value_options();
    $options = $this->value_options;
    $default_value = (array) $this->value;

    if (!empty($form_state['exposed'])) {
      $identifier = $this->options['expose']['identifier'];

      if (empty($this->options['expose']['use_operator']) || empty($this->options['expose']['operator'])) {
        // exposed and locked.
        $which = in_array($this->operator, $this->operator_values(1)) ? 'value' : 'none';
      }
      else {
        $source = 'edit-' . drupal_clean_css_identifier($this->options['expose']['operator']);
      }

      if (!empty($this->options['expose']['reduce'])) {
        $options = $this->reduce_value_options();

        if (empty($this->options['expose']['single']) && !empty($this->options['expose']['optional'])) {
          $default_value = array();
        }
      }

      if (!empty($this->options['expose']['single'])) {
        if (!empty($this->options['expose']['optional']) && (empty($default_value) || !empty($this->options['expose']['reduce']))) {
          $default_value = 'All';
        }
        else {
          if (empty($default_value)) {
            $keys = array_keys($options);
            $default_value = array_shift($keys);
          }
          else {
            $copy = $default_value;
            $default_value = array_shift($copy);
          }
        }
      }
    }

    $form['value'] = array(
      '#type' => 'select',
      '#title' => t('Country'),
      '#default_value' => $default_value,
      '#options' => $options,
      // Used by province autocompletion js.
      '#attributes' => array('class' => array('location_auto_country')),
      // Views will change this as necessary when exposed.
      '#multiple' => TRUE,
    );

    // Let location_autocomplete.js find the correct fields to attach.
    $form['value']['#attributes']['class'][] = 'location_auto_join_' . $this->options['expose']['identifier'];
  }

  public function reduce_value_options($input = NULL) {
    if (empty($this->options)) {
      $this->get_value_options();
    }
    if (!empty($this->options['expose']['reduce']) && !empty($this->options['value'])) {
      $reduced_options = array();
      foreach ($this->options['value'] as $value) {
        $reduced_options[$value] = $this->value_options[$value];
      }

      return $reduced_options;
    }

    return $this->get_value_options();
  }

  public function accept_exposed_input($input) {
    if (empty($this->options['exposed'])) {
      return TRUE;
    }

    if (!empty($this->options['expose']['use_operator']) && !empty($this->options['expose']['operator_id']) && isset($input[$this->options['expose']['operator_id']])) {
      $this->operator = $input[$this->options['expose']['operator_id']];
    }

    if (!empty($this->options['expose']['identifier'])) {
      $value = $input[$this->options['expose']['identifier']];

      if (empty($this->options['expose']['required'])) {
        if ($value == 'All' || $value === array()) {
          if (empty($this->options['value']) || (!empty($this->options['value']) && empty($this->options['expose']['reduce']))) {
            return FALSE;
          }
          else {
            $value = $this->options['value'];
          }
        }

        if (!empty($this->always_multiple) && $value === '') {
          return FALSE;
        }
      }

      if (isset($value)) {
        $this->value = $value;
        if (empty($this->always_multiple) && empty($this->options['expose']['multiple']) && !is_array($value)) {
          $this->value = array($value);
        }
      }
      else {
        return FALSE;
      }
    }

    return TRUE;
  }

  public function operator_options($which = 'title') {
    if (empty($this->options['expose']['multiple'])) {
      return array(
        'in' => t('Is'),
        'not in' => t('Is not'),
      );
    }
    else {
      return array(
        'in' => t('Is one of'),
        'not in' => t('Is not one of'),
      );
    }
  }

  public function query() {
    if (empty($this->value)) {
      return;
    }

    $this->ensure_my_table();
    $field = "$this->table_alias.$this->real_field";

    // Normalize values.
    $value = $this->value;
    if (is_array($value)) {
      if (count($value) == 1) {
        // If multiple is allowed but only one was chosen, use a string instead.
        $value = reset($value);
      }
    }

    if (is_array($value)) {
      // Multiple values
      $operator = ($this->operator == 'in') ? 'IN' : 'NOT IN';
      $this->query->add_where($this->options['group'], $field, $value, $operator);
    }
    else {
      // Single value
      $operator = ($this->operator == 'in') ? '=' : '!=';
      $this->query->add_where($this->options['group'], $field, $value, $operator);
    }
  }
}
